﻿using Sustainalytics.ClientDW.Entities;
using System;
using System.Collections.Generic;

namespace Sustainalytics.Entities
{
    public static class GCAssessmentExtensions
    {
        private static readonly Dictionary<GCAssessment, ReportName> _map = new Dictionary<GCAssessment, ReportName>
        {
            [GCAssessment.Watchlist] = ReportName.GCCS_Watchlist,
            [GCAssessment.NonCompliant] = ReportName.GCCS_NonCompliant,
        };

        public static string GetFullReportName(this GCAssessment assessment, string companyName, Guid companyId)
            => _map[assessment].GetFullReportName(companyName, companyId);

        public static string GetFullReportName(this GCAssessment assessment, string companyName, DateTime date)
            => _map[assessment].GetFullReportName(companyName, date);
    }
}
